#ifndef OP_COSINESIMILARITY_H
#define OP_COSINESIMILARITY_H

#include "operator_desc.h"
#include "op_runner.h"
#include "utils.h"

class OpCosineSimilarity {
public:
    OpCosineSimilarity()=default;

    ~OpCosineSimilarity();

    AtlasError Init();

    Result RunCosineSimilarityOp(vector<void*>& input, vector<size_t>& inputSize, vector<void*>& output);

private:
    void CreateOpDesc();
    Result SetInputData(vector<void*>& input, vector<size_t>& inputSize);

    OperatorDesc* opDesc_;
    OpRunner* opRunner_;
}; 



#endif //OP_COSINESIMILARITY_H
